home *** CD-ROM | disk | FTP | other *** search
/ Linux Cubed Series 7: Sunsite / Linux Cubed Series 7 - Sunsite Vol 1.iso / system / admin / linuxcon.000 / linuxcon / linuxconf-1.6 / fstab / partition.c < prev   
C/C++ Source or Header  |  1996-03-01  |  5KB  |  281 lines
  1. #include <stdio.h>
  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include "../misc/misc.h"
  5. #include "../netconf/netconf.h"
  6. #include "fstab.h"
  7. #include "fstab.m"
  8.  
  9. /*
  10.     describe one entry of the partition table
  11. */
  12. PUBLIC PARTITION::PARTITION(
  13.     const char *_dev,
  14.     int _id,
  15.     long _size)
  16. {
  17.     dev =strdup_err(_dev);
  18.     id = _id;
  19.     size = _size;
  20.     drive_letter = ' ';
  21. } 
  22.  
  23. PUBLIC PARTITION::~PARTITION()
  24. {
  25.     free (dev);
  26. }
  27.  
  28. /*
  29.     Set the drive letter DOS would normally associate with this
  30.     partition.
  31. */
  32. PUBLIC void PARTITION::setdosdrive(char letter)
  33. {
  34.     drive_letter = letter;
  35. }
  36. /*
  37.     Get the drive letter DOS would normally associate with this
  38.     partition.
  39. */
  40. PUBLIC char PARTITION::getdosdrive()
  41. {
  42.     return drive_letter;
  43. }
  44.  
  45. /*
  46.     Return != 0 if the partition is a DOS one.
  47. */
  48. PUBLIC int PARTITION::isdos()
  49. {
  50.     return id == PARTITION_DOS_ID12
  51.         || id == PARTITION_DOS_ID16
  52.         || id == PARTITION_DOS_ID1632;
  53. }
  54. /*
  55.     Return != 0 if the partition is a OS/2 one.
  56. */
  57. PUBLIC int PARTITION::isos2()
  58. {
  59.     return id == PARTITION_OS2_HPFS;
  60. }
  61.  
  62. /*
  63.     Return != if a partition is a swap.
  64. */
  65. PUBLIC int PARTITION::isswap()
  66. {
  67.     return id == PARTITION_LINUX_SWAP;
  68. }
  69. /*
  70.     Return != if a partition is a native linux one (ext2).
  71. */
  72. PUBLIC int PARTITION::islinux()
  73. {
  74.     /* #Specification: fsconf / partition / linux native
  75.         fsconf does not differentiate minix / xiafs / ext and
  76.         ext2 partition. It assumes that a "Linux native"
  77.         partition is always a ext2 one.
  78.     */
  79.     return id == PARTITION_LINUX_NATIVE;
  80. }
  81.  
  82.  
  83. /*
  84.     Return the path of the device controlling a partition.
  85. */
  86. PUBLIC const char *PARTITION::getdevice()
  87. {
  88.     return dev;
  89. }
  90.  
  91. /*
  92.     Return the size (in block) of the partition
  93. */
  94. PUBLIC long PARTITION::getsize()
  95. {
  96.     return size;
  97. }
  98.  
  99. /*
  100.     Return the ID (in block) of the partition
  101. */
  102. PUBLIC int PARTITION::getid()
  103. {
  104.     return id;
  105. }
  106.  
  107. PUBLIC void PARTITION::formatinfo(char *buf)
  108. {
  109.     char drive[3];
  110.     drive[0] = '\0';
  111.     if (drive_letter != ' '){
  112.         sprintf (drive,"%c:",drive_letter);
  113.     }
  114.     sprintf (buf,"%7ldM %-5s(%02x) %s"
  115.         ,size/1024,getos()
  116.         ,id,drive);
  117. }
  118.  
  119. /*
  120.     Return the name of the OS related to this partition if it is a
  121.     filesystem.
  122.  
  123.     Return "" is unknown
  124. */
  125. PUBLIC const char *PARTITION::getos()
  126. {
  127.     char *ret = "";
  128.     if (isdos()){
  129.         ret = "Dos";
  130.     }else if (islinux()){
  131.         ret = "Linux";
  132.     }else if (isswap()){
  133.         ret = "Swap";
  134.     }else if (isos2()){
  135.         ret = "Os/2";
  136.     }
  137.     return ret;
  138. }
  139.  
  140. PUBLIC PARTITIONS::PARTITIONS()
  141. {
  142.     DAEMON *dae = daemon_find ("fdisk");
  143.     if (dae != NULL){
  144.         char cmd[200];
  145.         sprintf (cmd,"%s -l",dae->getpath());
  146.         FILE *fin = popen (cmd,"r");
  147.         if (fin != NULL){
  148.             char buf[300];
  149.             while (fgets(buf,sizeof(buf)-1,fin)!=NULL){
  150.                 if(buf[0] == '/'){
  151.                     char t[7][30];
  152.                     int nb = sscanf (buf
  153.                         ,"%s %s %s %s %s %s %s"
  154.                         ,t[0],t[1],t[2],t[3],t[4],t[5]
  155.                         ,t[6]);
  156.                     if (nb != 7){
  157.                         xconf_error (
  158.                             MSG_U(E_IVLDFDISK
  159.                             ,"Invalid output format for fdisk\n%s")
  160.                             ,buf);
  161.                     }else{
  162.                         int offset = 4;
  163.                         if(strcmp(t[1],"*")==0) offset = 5;
  164.                         int id;
  165.                         sscanf (t[offset+1],"%x",&id);
  166.                         add (new PARTITION(t[0],id
  167.                             ,atoi(t[offset])));
  168.                     }
  169.                 }
  170.             }
  171.         }
  172.         pclose (fin);
  173.     }
  174.     setdosdrive();
  175. }
  176.  
  177. PUBLIC PARTITION *PARTITIONS::getitem(int no) const
  178. {
  179.     return (PARTITION*)ARRAY::getitem(no);
  180. }
  181.  
  182. /*
  183.     Locate one partition in the table using its device path as key.
  184.     Return NULL if not found.
  185. */
  186. PUBLIC PARTITION *PARTITIONS::getitem(const char *device) const
  187. {
  188.     PARTITION *ret = NULL;
  189.     for (int i=0; i<getnb(); i++){
  190.         PARTITION *p = getitem(i);
  191.         if (strcmp(p->getdevice(),device)==0){
  192.             ret = p;
  193.             break;
  194.         }
  195.     }
  196.     return ret;
  197. }
  198.  
  199. /*
  200.     Assign DOS drive letter to DOS partition
  201. */
  202. PROTECTED void PARTITIONS::setdosdrive()
  203. {
  204.     /* #Specification: partitions / dos drive / how it works
  205.         All this is very complicate.
  206.  
  207.         DOS allocate C on the first non extended DOS partition
  208.         of the first drive.
  209.  
  210.         It then allocate D if possible on the first non extended
  211.         DOS partition of the second drive.
  212.  
  213.         From there is get back to the first drive and allocate
  214.         a letter to all DOS extended partition. After that
  215.         it goes and finished the second hard drive.
  216.  
  217.         I have no idea what happen when you get more than
  218.         four drives. DOS does not support it if I am right.
  219.         Some info is needed here.
  220.     */
  221.     char drive = 'C';
  222.     char minpart = '1';
  223.     char maxpart = '4';
  224.     for (int pass=0; pass<2; pass++){
  225.         for (int d='a'; d<='b'; d++){
  226.             for (int i=minpart; i<=maxpart; i++){
  227.                 char buf[10];
  228.                 sprintf (buf,"/dev/hd%c%c",d,i);
  229.                 PARTITION *p = getitem (buf);
  230.                 if (p != NULL
  231.                     && p->isdos()){
  232.                     p->setdosdrive(drive);
  233.                     drive++;
  234.                     if (pass == 0) break;
  235.                 }
  236.             }
  237.         }
  238.         minpart = '5';
  239.         maxpart = '8';
  240.     }
  241. }
  242.  
  243. /*
  244.     Probe the partition table of all devices.
  245.     This take some time since fdisk is spawned to get it. So the
  246.     result is kept from call to call.
  247.     
  248.     The returned object should not be deleted.
  249. */
  250. PARTITIONS *partition_load()
  251. {
  252.     static PARTITIONS *parts = NULL;
  253.     if (parts == NULL){
  254.         parts = new PARTITIONS;
  255.     }
  256.     return parts;
  257. }
  258.  
  259. #ifdef TEST
  260.  
  261. int main (int , char *[])
  262. {
  263.     PARTITIONS parts;
  264.     for (int i=0; i<parts.getnb(); i++){
  265.         PARTITION *p = parts.getitem(i);
  266.         char *type = "???";
  267.         if (p->isswap()){
  268.             type = "Swap";
  269.         }else if (p->isdos()){
  270.             type = "Dos";
  271.         }else if (p->islinux()){
  272.             type = "Linux";
  273.         }
  274.         printf ("%s %ld %s\n",p->getdevice(),p->getsize()
  275.             ,type);
  276.     }
  277.     return 0;
  278. }
  279.  
  280. #endif
  281.